junyanz pytorch-cyclegan-and-pix2pix
junyanz/pytorch-CycleGAN-and-pix2pix
This is our ongoing PyTorch implementation for both unpaired and paired image-to-image translation. The code was written by Jun-Yan Zhu and Taesung Park. Check out the original CycleGAN Torch and pix2pix Torch code if you would like to reproduce the exact same results as in the papers. More example scripts can be found at scripts directory. To train a model on your own datasets, you need to create a data folder with two subdirectories trainA and trainB that contain images from domain A and B. You can test your model on your training set by setting phase'train' in test.lua.